cd ../

gpus=$1
exp_name="ids_bsz"

for ids_bsz in 48 72 96 128
do
    echo "Run experiments on <${exp_name}> with ${exp_name}=${ids_bsz}"
    echo "--gpus ${gpus}, --exp_name ${exp_name}/${ids_bsz}, --batch_sizes [0, 12, 12, ${ids_bsz}, 0]"
    python sed_main.py --gpus ${gpus}, --exp_name ${exp_name}/${ids_bsz} --batch_sizes "[0, 12, 12, ${ids_bsz}, 0]"
done